#导入蓝图模块
import re
from flask import Blueprint,request
from db import db1 
from flask.json import jsonify
import json 
#实例化
bp_course = Blueprint("course",__name__) 
#接口
@bp_course.route('/couponM',methods=['POST',"GET","PUT","DELETE"])
def couponM():
    #添加
    if request.method == "POST":
        #获取用户输入
        data = request.data
        if data:
            data = json.loads(data)
        else:
            data = request.form
        #构造sql语句
        sql = "insert into coupon values(0,'%s',%d,'%s','%s')"%(data['name'],int(data['total']),data['stime'],data['etime'])
        #pymysql
        db1.insert(sql)
        #返回结果
        return 'ok'
    elif request.method == 'GET':
        #根据id获取信息
        id = request.args.get('id')
        sql = "select * from coupon where id=%d"%(int(id))
        res = db1.find(sql)
        return jsonify({"code":200,'coupon':res})
    elif  request.method == "PUT":
        #修改
        #获取用户输入
        data = request.data
        if data:
            data = json.loads(data)
        else:
            data = request.form
            #构造Sql语句 update表名 set  字段名1=值 where id=1
            sql = "update coupon set name='%s',total=%d,starttime='%s',endtime='%s' where id=%d"%(data['name'],int(data['total']),data['stime'],data['etime'],int(data['id']))
            db1.insert(sql)
            return jsonify({'code':200})
    elif request.method == 'DELETE':
        #获取id
        id = request.args.get('id')
        sql = "delete from coupon where id=%d"%(int(id))
        db1.insert(sql)
        return jsonify({'code':200})

def get_data():
    data = request.data
    if data:
        data = json.loads(data)
    else:
        data = request.form
    return data
#部门管理
@bp_course.route('/deptM',methods=['POST','GET'])
def deptM():
    if request.method == "POST":
        data = get_data()
        #判断名字是否已经存在
        sql = "select id from dept where name='%s'"%(data['name'])
        res = db1.find(sql)
        #存在直接返回
        if res:
            return jsonify({"code":10011,'mes':'已经存在'})
        #不存在添加
        else:
            sql = "insert into dept(name) values('%s')"%(data['name'])
            db1.insert(sql)
            return jsonify({"code":200})
    else:
        sql = "select * from dept"
        res = db1.findall(sql)                 
        return jsonify({"code":200,'list':res})